# coding: utf-8
import numpy as np
import matplotlib.pylab as plt


def numerical_diff(f, x):#求导(直线)
    h = 1e-4 # 0.0001
    return (f(x+h) - f(x-h)) / (2*h)


def function_1(x):#曲线
    return 0.01*x**2 + 0.1*x 


def tangent_line(f, x):#tangent_line的意思是切线
    d = numerical_diff(f, x)#取得该点的导数值
    print("d=",d)
    y = f(x) - d*x#获取切线的截距
    return lambda t: d*t + y#返回一条切线的函数表达式
     
x = np.arange(0.0, 20.0, 0.1)#控制坐标轴范围
y = function_1(x)
plt.xlabel("x")
plt.ylabel("f(x)")

tf = tangent_line(function_1, 5)#绘制x=5的切线
y2 = tf(x)

plt.plot(x, y)
plt.plot(x, y2)
plt.show()
